import React from 'react';
import { observer, inject } from 'mobx-react';
import { Button, Space } from 'antd';
import { StoreProps } from '../store';
import styles from '../style/features.module.css';

const Polish: React.FC<StoreProps> = ({ documentStore, proofreadStore }) => {
  const handlePolish = () => {
    documentStore.polishContent();
  };

  return (
    <div className={styles['statHub-feature-container']}>
      <div className={styles['statHub-feature-actions']}>
        <Space>
          <Button type="primary" onClick={handlePolish} loading={documentStore.processing}>
            一键润色
          </Button>
        </Space>
      </div>
    </div>
  );
};

export default inject('documentStore', 'proofreadStore')(observer(Polish)); 